AI-assisted registration